// test.cpp : Defines the entry point for the application.
//

#include "stdafx.h"
#include "test.h"
#include <commctrl.h>
#pragma comment(lib,"comctl32.lib")
#include <shlobj.h>
#include <shellapi.h>
#include <WindowsX.h>

#pragma  comment(lib,"WS2_32.LIB")
#include <winsock2.h>
#include <atlconv.h>
#include <concrtrm.h>

#include <process.h>
#include "ServerProc.h"



#define MAX_LOADSTRING 100


// Global Variables:
HINSTANCE hInst;								// current instance
TCHAR szTitle[MAX_LOADSTRING];					// The title bar text
TCHAR szWindowClass[MAX_LOADSTRING];			// the main window class name

// Forward declarations of functions included in this code module:
ATOM				MyRegisterClass(HINSTANCE hInstance);
BOOL				InitInstance(HINSTANCE, int);
LRESULT CALLBACK	WndProc(HWND, UINT, WPARAM, LPARAM);
INT_PTR CALLBACK	About(HWND, UINT, WPARAM, LPARAM);

int APIENTRY _tWinMain(HINSTANCE hInstance,
                     HINSTANCE hPrevInstance,
                     LPTSTR    lpCmdLine,
                     int       nCmdShow)
{
	UNREFERENCED_PARAMETER(hPrevInstance);
	UNREFERENCED_PARAMETER(lpCmdLine);

 	// TODO: Place code here.
	MSG msg;
	HACCEL hAccelTable;

	// Initialize global strings
	LoadString(hInstance, IDS_APP_TITLE, szTitle, MAX_LOADSTRING);
	LoadString(hInstance, IDC_TEST, szWindowClass, MAX_LOADSTRING);
	MyRegisterClass(hInstance);

	// Perform application initialization:
	if (!InitInstance (hInstance, nCmdShow))
	{
		return FALSE;
	}

	hAccelTable = LoadAccelerators(hInstance, MAKEINTRESOURCE(IDC_TEST));

	// Main message loop:
	while (GetMessage(&msg, NULL, 0, 0))
	{
		if (!TranslateAccelerator(msg.hwnd, hAccelTable, &msg))
		{
			TranslateMessage(&msg);
			DispatchMessage(&msg);
		}
	}

	return (int) msg.wParam;
}



//
//  FUNCTION: MyRegisterClass()
//
//  PURPOSE: Registers the window class.
//
//  COMMENTS:
//
//    This function and its usage are only necessary if you want this code
//    to be compatible with Win32 systems prior to the 'RegisterClassEx'
//    function that was added to Windows 95. It is important to call this function
//    so that the application will get 'well formed' small icons associated
//    with it.
//
ATOM MyRegisterClass(HINSTANCE hInstance)
{
	WNDCLASSEX wcex;

	wcex.cbSize = sizeof(WNDCLASSEX);

	wcex.style			= CS_HREDRAW | CS_VREDRAW;
	wcex.lpfnWndProc	= WndProc;
	wcex.cbClsExtra		= 0;
	wcex.cbWndExtra		= 0;
	wcex.hInstance		= hInstance;
	wcex.hIcon			= LoadIcon(hInstance, MAKEINTRESOURCE(IDI_TEST));
	wcex.hCursor		= LoadCursor(NULL, IDC_ARROW);
	wcex.hbrBackground	= (HBRUSH)(COLOR_WINDOW+1);
	wcex.lpszMenuName	= MAKEINTRESOURCE(IDC_TEST);
	wcex.lpszClassName	= szWindowClass;
	wcex.hIconSm		= LoadIcon(wcex.hInstance, MAKEINTRESOURCE(IDI_SMALL));

	return RegisterClassEx(&wcex);
}

//
//   FUNCTION: InitInstance(HINSTANCE, int)
//
//   PURPOSE: Saves instance handle and creates main window
//
//   COMMENTS:
//
//        In this function, we save the instance handle in a global variable and
//        create and display the main program window.
//
HWND hCBBox ;
BOOL InitInstance(HINSTANCE hInstance, int nCmdShow)
{
   HWND hWnd;

   hInst = hInstance; // Store instance handle in our global variable

   hWnd = CreateWindow(szWindowClass, szTitle, WS_OVERLAPPEDWINDOW,
      CW_USEDEFAULT, 0, CW_USEDEFAULT, 0, NULL, NULL, hInstance, NULL);



   if (!hWnd)
   {
      return FALSE;
   }

   ShowWindow(hWnd, nCmdShow);
   UpdateWindow(hWnd);

   return TRUE;
}

//
//  FUNCTION: WndProc(HWND, UINT, WPARAM, LPARAM)
//
//  PURPOSE:  Processes messages for the main window.
//
//  WM_COMMAND	- process the application menu
//  WM_PAINT	- Paint the main window
//  WM_DESTROY	- post a quit message and return
//
//
LPTSTR Convert(char *str, int len)
{
	
	LPTSTR dest = new TCHAR[len+1];
	int i;
	for ( i = 0 ; i < len && str[i]!='\0' ; i++)
	{
		dest[i] = (TCHAR)str[i];
	}
	dest[i] = '\0';
	return dest;
}
HWND hButton;
HWND hEditNumMem;
void thread1(PVOID hWnd)
{
	WSADATA wsadata;
	WSAStartup(MAKEWORD(2,0),&wsadata); //return = 0 -> successful !
	SOCKET s;
	struct sockaddr_in sai,csai; //sai :server address_in , csai: client sockaddr_in
	int sockadr_in;
	//socket
	s = socket(AF_INET,SOCK_STREAM,0); // AF_INET: ket noi giua nhieu may --- SOCK_STREAM: tcp --- vi` dung` sock_stream nen de protocols = 0
	//bind

	sai.sin_family = AF_INET;
	sai.sin_port = 1234;
	sai.sin_addr.s_addr = inet_addr("127.0.0.1"); 

	bind(s,(sockaddr*)&sai,sizeof(sai));
	//listen
	TCHAR mem[2];
	GetWindowText((HWND)hEditNumMem, mem,2);
	SetWindowText((HWND)hWnd,mem);
	int numMem = _wtoi(mem);
	//Tao mang socket
	SOCKET *c = new SOCKET[numMem];
	//Array nickname
	char** nick;
	nick = new char*[numMem];
	char** buff;
	buff = new char*[numMem];
	for(int i = 0; i < numMem; i++)
	{
		nick[i] = new char[100];
		buff[i] = new char[100];
	}
	int k = 0;
	listen(s,numMem);
	EnableWindow(hEditNumMem,FALSE);
	sockadr_in = sizeof(csai);
	
	for(int i = 0 ; i < numMem ;i++)
	{
		c[i] = accept(s,(sockaddr *)&csai, &sockadr_in);
	}
	//Anounce client can register nickname and process nickname
	for(int i = 0 ; i < numMem;i++)
	{
		int number = 1;
		ServerProc p;
		send(c[i],(char*)&number,sizeof(int),0);
		recv(c[i],buff[i],100,0);

		buff[i][strlen(buff[i])] = '\0';

		bool temp = p.nickNameProc(buff[i],nick,i);
		number = (int)temp * (i+1);
		send(c[i],(char*)&number,sizeof(int),MSG_DONTROUTE);
		if(temp == true)
		{
			nick[k] = buff[i];
			k++;
		}
		else
		{
			i--;
		}
	//
	shutdown(c[i],SD_BOTH);
	closesocket(c[i]);
	}
	
	closesocket(s);
	WSACleanup();
	delete[] c;
	for(int i = 0; i < numMem; i++)
	{
		nick[i] = NULL;
		buff[i] = NULL;
	}
	nick = NULL;
	buff = NULL;
}
BOOL OnCreate(HWND hwnd, LPCREATESTRUCT lpCreateStruct)
{
	hButton = CreateWindow(_T("BUTTON"),L"server",WS_CHILD | WS_VISIBLE | BS_PUSHBUTTON,50,50,50,50,hwnd,(HMENU)ID_BUTTON,hInst,NULL);	
	hEditNumMem = CreateWindow(_T("EDIT"),NULL,WS_CHILD|WS_VISIBLE|ES_NUMBER|WS_BORDER,25,25,50,25,hwnd,NULL,hInst,NULL);
	
	return TRUE;
	
}
LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
	int wmId, wmEvent;
	PAINTSTRUCT ps;
	HDC hdc;
	
	switch (message)
	{
	HANDLE_MSG(hWnd,WM_CREATE,OnCreate);
	
	case WM_COMMAND:
		wmId    = LOWORD(wParam);
		wmEvent = HIWORD(wParam);
		// Parse the menu selections:
		switch (wmId)
		{
	
		case ID_BUTTON:
			{
				EnableWindow(hButton,FALSE);
				_beginthread(thread1,0,hWnd);
				
				//delete buf;
				
			}break;
		case IDM_ABOUT:
			DialogBox(hInst, MAKEINTRESOURCE(IDD_ABOUTBOX), hWnd, About);
			break;
		case IDM_EXIT:
			DestroyWindow(hWnd);
			break;
		default:
			return DefWindowProc(hWnd, message, wParam, lParam);
		}
		break;
	case WM_PAINT:
		hdc = BeginPaint(hWnd, &ps);
		// TODO: Add any drawing code here...
		EndPaint(hWnd, &ps);
		break;
	case WM_DESTROY:
		PostQuitMessage(0);
		break;
	default:
		return DefWindowProc(hWnd, message, wParam, lParam);
	}
	return 0;
}

// Message handler for about box.
INT_PTR CALLBACK About(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam)
{
	UNREFERENCED_PARAMETER(lParam);
	switch (message)
	{
	case WM_INITDIALOG:
		return (INT_PTR)TRUE;

	case WM_COMMAND:
		if (LOWORD(wParam) == IDOK || LOWORD(wParam) == IDCANCEL)
		{
			EndDialog(hDlg, LOWORD(wParam));
			return (INT_PTR)TRUE;
		}
		break;
	}
	return (INT_PTR)FALSE;
}
